package veterinaria.dao;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collection;

import veterinaria.excepcion.DAOExcepcion;
import veterinaria.modelo.Medicamento;
import veterinaria.util.ConexionBD;

public class MedicamentoDAO extends BaseDAO {

	public Collection<Medicamento> buscarPorNombre(String nombre)
			throws DAOExcepcion {
		String query = "select codigo, nombre, descripcion from Medicamento where nombre like ?";
		Collection<Medicamento> lista = new ArrayList<Medicamento>();
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setString(1, "%" + nombre + "%");
			rs = stmt.executeQuery();
			while (rs.next()) {
				Medicamento vo = new Medicamento();
				vo.setCodigo(rs.getInt("id_Medicamento"));
				vo.setNombre(rs.getString("nombre"));
				vo.setDescripcion(rs.getString("descripcion"));
				lista.add(vo);
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		System.out.println(lista.size());
		return lista;
	}

	public Medicamento insertar(Medicamento vo) throws DAOExcepcion {
		String query = "insert into medicamento(nombre,precio,cantidad,descripcion,imagen) values (?,?,?,?,?)";
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			
			
			stmt.setString(1, vo.getNombre());
			stmt.setDouble(2, vo.getPrecio());
			stmt.setInt(3, vo.getCantidad());
			stmt.setString(4, vo.getDescripcion());
			stmt.setString(5, vo.getImagen());
			
			int i = stmt.executeUpdate();
			if (i != 1) {
				throw new SQLException("No se pudo insertar");
			}
			
			// Obtener el ultimo id
						int id = 0;
						query = "select last_insert_id()";
						stmt = con.prepareStatement(query);
						rs = stmt.executeQuery();
						if (rs.next()) {
							id = rs.getInt(1);
						}
						vo.setCodigo(id);

		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
		
	}

	public Medicamento obtener(int a) throws DAOExcepcion {
		Medicamento vo = new Medicamento();
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			String query = "select idmedicamento,nombre,precio,cantidad,descripcion from medicamento where idMedicamento=?";
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setInt(1, a);
			rs = stmt.executeQuery();
			if (rs.next()) {
				vo.setCodigo(rs.getInt(1));
				vo.setNombre(rs.getString(2));
				vo.setPrecio(rs.getDouble(3));
				vo.setCantidad(rs.getInt(4));
				vo.setDescripcion(rs.getString(5));
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
	}

	public Medicamento obtener2(String atributo,Object param) throws DAOExcepcion {
		Medicamento vo = new Medicamento();
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			String query ="select idMedicamento, nombre, precio, cantidad, descripcion, imagen from medicamento where ";
			
			if ("CODIGO".equals(atributo))
				query = query + "idMedicamento=?";
			if ("NOMBRE".equals(atributo))
				query = query + "nombre=?";
			
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);

			if ("CODIGO".equals(atributo))
				stmt.setInt(1, (int)param);
			if ("NOMBRE".equals(atributo))
				stmt.setString(1, (String)param);
			
			rs = stmt.executeQuery();
			if (rs.next()) {
				vo.setCodigo(rs.getInt(1));
				vo.setNombre(rs.getString(2));
				vo.setPrecio(rs.getDouble(3));
				vo.setCantidad(rs.getInt(4));
				vo.setDescripcion(rs.getString(5));
				vo.setImagen(rs.getString(6));
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
	}
	
	public void eliminar(int a) throws DAOExcepcion {
		String query = "delete from medicamento WHERE idMedicamento=?";
		Connection con = null;
		PreparedStatement stmt = null;
		try {
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setInt(1, a);
			int i = stmt.executeUpdate();
			if (i != 1) {
				throw new SQLException("No se pudo eliminar");
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
	}

	public Medicamento actualizar(Medicamento vo) throws DAOExcepcion {
		String query = "update medicamento set nombre=?,precio=?,cantidad=?,descripcion=? where idMedicamento=?";
		Connection con = null;
		PreparedStatement stmt = null;
		try {
			con = ConexionBD.obtenerConexion();
			stmt = con.prepareStatement(query);
			stmt.setString(1, vo.getNombre());
			stmt.setDouble(2, vo.getPrecio());
			stmt.setInt(3, vo.getCantidad());
			stmt.setString(4, vo.getDescripcion());
			stmt.setInt(5, vo.getCodigo());
			int i = stmt.executeUpdate();
			if (i != 1) {
				throw new SQLException("No se pudo actualizar");
			}
		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return vo;
	}

	public Collection<Medicamento> listar() throws DAOExcepcion {
		Collection<Medicamento> c = new ArrayList<Medicamento>();
		Connection con = null;
		PreparedStatement stmt = null;
		ResultSet rs = null;
		try {
			con = ConexionBD.obtenerConexion();
			String query = "select idmedicamento,nombre,precio,cantidad,descripcion from medicamento order by nombre";
			stmt = con.prepareStatement(query);
			rs = stmt.executeQuery();
			while (rs.next()) {
				Medicamento vo = new Medicamento();
				vo.setCodigo(rs.getInt("idMedicamento"));
				vo.setNombre(rs.getString("nombre"));
				vo.setPrecio(rs.getDouble("precio"));
				vo.setCantidad(rs.getInt("cantidad"));
				vo.setDescripcion(rs.getString("descripcion"));
				
				c.add(vo);
			}

		} catch (SQLException e) {
			System.err.println(e.getMessage());
			throw new DAOExcepcion(e.getMessage());
		} finally {
			this.cerrarResultSet(rs);
			this.cerrarStatement(stmt);
			this.cerrarConexion(con);
		}
		return c;
	}

}
